package cn.learn.mq.rabbitmq.work_queue;

import cn.learn.mq.rabbitmq.RabbitMqConnectionUtil;
import cn.learn.mq.rabbitmq.SleepUtil;
import com.rabbitmq.client.CancelCallback;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.DeliverCallback;
import org.springframework.util.Assert;

import java.io.IOException;

/**
 * 改成手动应答
 *
 * @author huangyezhan
 * @version 1.0
 * @date 2021年08月22日8:41
 */
public class WorkAutoAck01 {
    private static String queue_name = "work_queue_name";

    public static void main(String[] args) throws IOException {
        Channel channel = RabbitMqConnectionUtil.creatConnection();
        Assert.notNull(channel,"信道不能为空！");
        //推送的消息如何进行消费的接口回调
        DeliverCallback deliverCallback = (consumerTag, delivery) -> {
            String message = new String(delivery.getBody());
            SleepUtil.sleepUtil(1);
            // 1、消息标记tag；2、false代表只应答接收到的那个传递的消息，true为应答信道上的所有消息
            channel.basicAck(delivery.getEnvelope().getDeliveryTag(),false);
            System.out.println("接收到消息：" + message);
        };
        //取消消费的一个回调接口 如在消费的时候队列被删除掉了
        CancelCallback cancelCallback = (consumerTag) -> {
            System.out.println("消费者取消消费接口回调逻辑");
        };
        System.out.println("work01消费者等待消费···");
        /*
         * 消费者消费消息
         * 1.消费哪个队列
         * 2.消费成功之后是否要自动应答 true 代表自动应答， false 手动应答
         * 3.消费者成功 消费的回调
         * 4.消费者取消 消费的回调
         */
        // 设置成不公平分发
        channel.basicQos(1);
        // 手动应答
        boolean autoAck = false;
        channel.basicConsume(queue_name, autoAck, deliverCallback, cancelCallback);
    }
}

class WorkAutoAck02 {
    private static String queue_name = "work_queue_name";

    public static void main(String[] args) throws IOException {
        Channel channel = RabbitMqConnectionUtil.creatConnection();
        Assert.notNull(channel,"信道不能为空！");
        //推送的消息如何进行消费的接口回调
        DeliverCallback deliverCallback = (consumerTag, delivery) -> {
            String message = new String(delivery.getBody());
            SleepUtil.sleepUtil(30);
            // 1、消息标记tag；2、false代表只应答接收到的那个传递的消息，true为应答信道上的所有消息
            channel.basicAck(delivery.getEnvelope().getDeliveryTag(),false);
            System.out.println("接收到消息：" + message);
        };
        //取消消费的一个回调接口 如在消费的时候队列被删除掉了
        CancelCallback cancelCallback = (consumerTag) -> {
            System.out.println("消费者取消消费接口回调逻辑");
        };
        System.out.println("work02消费者等待消费···");
        channel.basicQos(1);
        // 手动应答
        boolean autoAck = false;
        channel.basicConsume(queue_name, autoAck, deliverCallback, cancelCallback);
    }
}
